package com.leps.nq.sys.service.impl;

import cn.dev33.satoken.session.SaSession;
import cn.dev33.satoken.stp.StpInterface;
import cn.dev33.satoken.stp.StpUtil;
import com.leps.nq.sys.entity.SysRole;
import com.leps.nq.sys.mapper.SysMenuMapper;
import com.leps.nq.sys.mapper.SysRoleMapper;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;

import java.util.List;
import java.util.stream.Collectors;

/**
 * 自定义权限验证接口扩展
 */
@Component    // 保证此类被SpringBoot扫描，完成Sa-Token的自定义权限验证扩展
public class StpInterfaceImpl implements StpInterface {
    @Autowired
    private SysMenuMapper menuMapper;
    @Autowired
    private SysRoleMapper roleMapper;

    /**
     * 返回一个账号所拥有的权限码集合
     */
    public List<String> selectPermissions(Object loginId, String loginType) {
        final Long userId = Long.valueOf(String.valueOf(loginId));
        return this.menuMapper.selectMenuPermsByUserId(userId);
    }

    /**
     * 返回一个账号所拥有的角色标识集合 (权限与角色可分开校验)
     */
    public List<String> selectRoles(Object loginId, String loginType) {
        final Long userId = Long.valueOf(loginId.toString());
        List<SysRole> roles = this.roleMapper.selectRolesByUserId(userId);
        return roles.stream().map(SysRole::getRoleKey).collect(Collectors.toList());
    }


    // 返回一个账号所拥有的权限码集合
    @Override
    public List<String> getPermissionList(Object loginId, String loginType) {
        SaSession session = StpUtil.getSessionByLoginId(loginId);
        return session.get("Permission_List", () -> {
            return  this.selectPermissions(loginId, loginType);
        });
//        // 1. 声明权限码集合
//        List<String> permissionList = new ArrayList<>();
//
//        // 2. 遍历角色列表，查询拥有的权限码
//        for (String roleId : getRoleList(loginId, loginType)) {
//            SaSession roleSession = SaSessionCustomUtil.getSessionById("role-" + roleId);
//            List<String> list = roleSession.get("Permission_List", () -> {
//                return ...;     // 从数据库查询这个角色所拥有的权限列表
//            });
//            permissionList.addAll(list);
//        }

//        // 3. 返回权限码集合
//        return permissionList;
    }

    // 返回一个账号所拥有的角色标识集合
    @Override
    public List<String> getRoleList(Object loginId, String loginType) {
        SaSession session = StpUtil.getSessionByLoginId(loginId);
        return session.get("Role_List", () -> {
            return  this.selectRoles(loginId, loginType);
        });
    }


}

